package com.xqiapp.jsp.config;

import com.alibaba.druid.support.http.StatViewServlet;
import com.alibaba.druid.support.http.WebStatFilter;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.boot.web.servlet.FilterRegistrationBean;
import org.springframework.boot.web.servlet.ServletRegistrationBean;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.jdbc.datasource.DataSourceTransactionManager;
import org.springframework.scheduling.annotation.EnableScheduling;
import org.springframework.transaction.PlatformTransactionManager;

import javax.sql.DataSource;
import java.util.HashMap;
import java.util.Map;

/**
 * Created by rwrwd7 on 2017/9/21.
 */
@Configuration
@EnableScheduling
public class DataSourceConfig {

    @Bean(name = "dataSource")
    public DataSource dataSource() {
        //按照目标数据源名称和目标数据源对象的映射存放在Map中
        Map<Object, Object> targetDataSources = new HashMap<>();
        //查找xml数据连接字符串
        targetDataSources = getTargetDataSources();
        //采用是想AbstractRoutingDataSource的对象包装多数据源
        DataSourceDynamic dataSource = new DataSourceDynamic();
        dataSource.setTargetDataSources(targetDataSources);
        //设置默认的数据源，当拿不到数据源时，使用此配置
        //dataSource.setDefaultTargetDataSource(properties.getUzaiTravel());
        return dataSource;
    }
    @Bean
    public PlatformTransactionManager txManager() {
        return new DataSourceTransactionManager(dataSource());
    }

    public static Object[] targetTypeNames = {"master","ds1","ds2"};


    /**
     *获取数据源集合
     */
    private Map<Object, Object> getTargetDataSources() {
        try {
            Map<Object, Object> targetDataSources = new HashMap<>();
            for(int i = 0 ; i < targetTypeNames.length ; i ++) {
                targetDataSources.put(targetTypeNames[i], DataSourceFactory.createDataSource(targetTypeNames[i]));
            }
            //targetDataSources.put(targetTypeNames[1], ds1DataSource);
            return targetDataSources;
        } catch (Exception ex) {
            return null;
        }
    }

    //访问的ip
    @Value("${druid.IP}")
    private String IP;
    //登录名
    @Value("${druid.druidLgoinName}")
    private String druidLgoinName;
    //密码
    @Value("${druid.druidLgoinPassword}")
    private String druidLgoinPassword;
    @Bean
    public ServletRegistrationBean DruidStatViewServle() {
        //org.springframework.boot.context.embedded.ServletRegistrationBean提供类的进行注册.
        ServletRegistrationBean servletRegistrationBean = new ServletRegistrationBean(new StatViewServlet(), "/druid/*");
        //添加初始化参数：initParams
        //白名单：
        servletRegistrationBean.addInitParameter("allow",IP);
        //IP黑名单 (存在共同时，deny优先于allow) : 如果满足deny的话提示:Sorry, you are not permitted to view this page.
        // servletRegistrationBean.addInitParameter("deny", "192.168.1.73");
        //登录查看信息的账号密码.
        servletRegistrationBean.addInitParameter("loginUsername",druidLgoinName);
        servletRegistrationBean.addInitParameter("loginPassword",druidLgoinPassword);
        //是否能够重置数据.
        servletRegistrationBean.addInitParameter("resetEnable","false");
        return servletRegistrationBean;
    }
    /**
     * 注册一个：filterRegistrationBean
     * @return
     */
    @Bean
    public FilterRegistrationBean druidStatFilter2(){
        FilterRegistrationBean filterRegistrationBean = new FilterRegistrationBean(new WebStatFilter());
        //添加过滤规则.
        filterRegistrationBean.addUrlPatterns("/*");
        //添加不需要忽略的格式信息.
        filterRegistrationBean.addInitParameter("exclusions","*.js,*.gif,*.jpg,*.png,*.css,*.ico,/druid/*");
        return filterRegistrationBean;
    }
}
